"""
    step function 阶跃函数
"""
import numpy as np
import matplotlib.pylab as plt

def step_function(x):
    # if x < 0:
    #     return 0
    # else:
    #     return 1
    return np.array(x > 0,dtype=np.int_)


if __name__ == '__main__':
    x = np.arange(-5, 5, 0.01)
    y = step_function(x)
    plt.plot(x, y)
    # 指定 y 轴的范围
    plt.ylim(-0.1, 1.1)
    plt.show()